package sys

import (
	"github.com/gofiber/fiber/v2"
	"go-web2/app/common/config"
	"go-web2/app/model/sys"
)

type SafeController struct{}

// 获取安全设置
func (SafeController) GetSafeSet(c *fiber.Ctx) error {
	safe := sys.SysSafe{}
	safe.GetById()
	return c.Status(200).JSON(config.Success(safe))
}

// 修改安全设置
func (SafeController) Update(c *fiber.Ctx) error {
	var safe sys.SysSafe
	if err := c.BodyParser(&safe); err != nil {
		return c.Status(200).JSON(config.Error(err.Error()))
	}
	safe.Token = c.Get(config.TokenHeader)
	err := safe.Update()
	if err != nil {
		return c.Status(200).JSON(config.Error(err.Error()))
	}
	return c.Status(200).JSON(config.Success(nil))
}
